package com.thread.thread09;

import java.util.LinkedList;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

public class MyQueue {
	// 定义一个对列
	private LinkedList<Object> linkedList = new LinkedList<>();
	// 定义一个计数器
	private final AtomicInteger count = new AtomicInteger(0);

	// 定义上限和下限
	private final int minSize = 0;
	private final int maxSize;

	public MyQueue(int maxSize) {
		this.maxSize = maxSize;
	}

	// 5 初始化一个对象 用于加锁
	private final Object lock = new Object();

	public void put(Object obj) {
		synchronized (lock) {
			while (count.get() == maxSize) {
				try {
					lock.wait();
				} catch (InterruptedException e) {
					// TODO Auto-generated catch block
					e.printStackTrace();
				}
			}
			linkedList.add(obj);
			count.incrementAndGet();
			lock.notify();
			System.out.println("新加入的元素是 " + obj);
		}
	}

	public Object take() {
		Object ret = null;
		synchronized (lock) {
			while (count.get() == this.minSize) {
				try {
					lock.wait();
				} catch (InterruptedException e) {
					// TODO Auto-generated catch block
					e.printStackTrace();
				}
			}
			ret = linkedList.removeFirst();
			count.decrementAndGet();
			lock.notify();
		}
		return ret;
	}

	public int getSize() {
		return this.count.get();
	}
	
	public static void main(String[] args) {
		final MyQueue mq = new MyQueue(5);
		mq.put("a");
		mq.put("b");
		mq.put("c");
		mq.put("d");
		mq.put("e");
		System.out.println("当前容器的长度:" + mq.getSize());
		Thread t1 = new Thread(new Runnable() {
			@Override
			public void run() {
				mq.put("f");
				mq.put("g");
			}
		},"t1");
		
		t1.start();
		
		Thread t2 = new Thread(new Runnable() {
			@Override
			public void run() {
				Object o1 = mq.take();
				System.out.println("移除的元素为:" + o1);
				Object o2 = mq.take();
				System.out.println("移除的元素为:" + o2);
			}
		},"t2");
		

		try {
			TimeUnit.SECONDS.sleep(2);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		
		t2.start();
	}
}
